home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / share / pyshared / PIL / WmfImagePlugin.py < prev    next >
Text File  |  2006-12-03  |  4KB  |  149 lines
  1. #
  2. # The Python Imaging Library
  3. # $Id: WmfImagePlugin.py 2134 2004-10-06 08:55:20Z fredrik $
  4. #
  5. # WMF stub codec
  6. #
  7. # history:
  8. # 1996-12-14 fl   Created
  9. # 2004-02-22 fl   Turned into a stub driver
  10. # 2004-02-23 fl   Added EMF support
  11. #
  12. # Copyright (c) Secret Labs AB 1997-2004.  All rights reserved.
  13. # Copyright (c) Fredrik Lundh 1996.
  14. #
  15. # See the README file for information on usage and redistribution.
  16. #
  17.  
  18. __version__ = "0.2"
  19.  
  20. import Image, ImageFile
  21.  
  22. _handler = None
  23.  
  24. ##
  25. # Install application-specific WMF image handler.
  26. #
  27. # @param handler Handler object.
  28.  
  29. def register_handler(handler):
  30.     global _handler
  31.     _handler = handler
  32.  
  33. # --------------------------------------------------------------------
  34.  
  35. def word(c, o=0):
  36.     return ord(c[o]) + (ord(c[o+1])<<8)
  37.  
  38. def short(c, o=0):
  39.     v = ord(c[o]) + (ord(c[o+1])<<8)
  40.     if v >= 32768:
  41.         v = v - 65536
  42.     return v
  43.  
  44. def dword(c, o=0):
  45.     return ord(c[o]) + (ord(c[o+1])<<8) + (ord(c[o+2])<<16) + (ord(c[o+3])<<24)
  46.  
  47. def long(c, o=0):
  48.     return dword(c, o)
  49.  
  50. #
  51. # --------------------------------------------------------------------
  52. # Read WMF file
  53.  
  54. def _accept(prefix):
  55.     return (
  56.         prefix[:6] == "\xd7\xcd\xc6\x9a\x00\x00" or
  57.         prefix[:4] == "\x01\x00\x00\x00"
  58.         )
  59.  
  60. ##
  61. # Image plugin for Windows metafiles.
  62.  
  63. class WmfStubImageFile(ImageFile.StubImageFile):
  64.  
  65.     format = "WMF"
  66.     format_description = "Windows Metafile"
  67.  
  68.     def _open(self):
  69.  
  70.         # check placable header
  71.         s = self.fp.read(80)
  72.  
  73.         if s[:6] == "\xd7\xcd\xc6\x9a\x00\x00":
  74.  
  75.             # placeable windows metafile
  76.  
  77.             # get units per inch
  78.             inch = word(s, 14)
  79.  
  80.             # get bounding box
  81.             x0 = short(s, 6); y0 = short(s, 8)
  82.             x1 = short(s, 10); y1 = short(s, 12)
  83.  
  84.             # normalize size to 72 dots per inch
  85.             size = (x1 - x0) * 72 / inch, (y1 - y0) * 72 / inch
  86.  
  87.             self.info["wmf_bbox"] = x0, y0, x1, y1
  88.  
  89.             self.info["dpi"] = 72
  90.  
  91.             # print self.mode, self.size, self.info
  92.  
  93.             # sanity check (standard metafile header)
  94.             if s[22:26] != "\x01\x00\t\x00":
  95.                 raise SyntaxError("Unsupported WMF file format")
  96.  
  97.         elif long(s) == 1 and s[40:44] == " EMF":
  98.             # enhanced metafile
  99.  
  100.             # get bounding box
  101.             x0 = long(s, 8); y0 = long(s, 12)
  102.             x1 = long(s, 16); y1 = long(s, 20)
  103.  
  104.             # get frame (in 0.01 millimeter units)
  105.             frame = long(s, 24), long(s, 28), long(s, 32), long(s, 36)
  106.  
  107.             # normalize size to 72 dots per inch
  108.             size = x1 - x0, y1 - y0
  109.  
  110.             # calculate dots per inch from bbox and frame
  111.             xdpi = 2540 * (x1 - y0) / (frame[2] - frame[0])
  112.             ydpi = 2540 * (y1 - y0) / (frame[3] - frame[1])
  113.  
  114.             self.info["wmf_bbox"] = x0, y0, x1, y1
  115.  
  116.             if xdpi == ydpi:
  117.                 self.info["dpi"] = xdpi
  118.             else:
  119.                 self.info["dpi"] = xdpi, ydpi
  120.  
  121.         else:
  122.             raise SyntaxError("Unsupported file format")
  123.  
  124.         self.mode = "RGB"
  125.         self.size = size
  126.  
  127.         loader = self._load()
  128.         if loader:
  129.             loader.open(self)
  130.  
  131.     def _load(self):
  132.         return _handler
  133.  
  134.  
  135. def _save(im, fp, filename):
  136.     if _handler is None or not hasattr("_handler", "save"):
  137.         raise IOError("WMF save handler not installed")
  138.     _handler.save(im, fp, filename)
  139.  
  140. #
  141. # --------------------------------------------------------------------
  142. # Registry stuff
  143.  
  144. Image.register_open(WmfStubImageFile.format, WmfStubImageFile, _accept)
  145. Image.register_save(WmfStubImageFile.format, _save)
  146.  
  147. Image.register_extension(WmfStubImageFile.format, ".wmf")
  148. Image.register_extension(WmfStubImageFile.format, ".emf")
  149.